package com.example.lotterysystem.common.utils;

/**
 * @auther falls_vc
 * description:
 * @date 2025/5/14  20:55
 */

import cn.hutool.captcha.LineCaptcha;
import cn.hutool.captcha.generator.RandomGenerator;

/**
 * 验证码
 */
public class CaptchaUtil {
    /**
     * 获取指定长度 纯数字验证码
     *
     * @param length
     * @return
     */
    public static String getCaptcha(int length) {
        // 自定义纯数字的验证码（随机4位数字，可重复）
        RandomGenerator randomGenerator = new RandomGenerator("0123456789", length);
        LineCaptcha lineCaptcha = cn.hutool.captcha.CaptchaUtil.createLineCaptcha(200, 100);
        lineCaptcha.setGenerator(randomGenerator);
// 重新生成code
        lineCaptcha.createCode();

        return lineCaptcha.getCode();
    }
}
